672a8990090608772c05c92c34d9505e99f3e73d,cassandra-scheduler/src/test/java/com/mesosphere/dcos/cassandra/scheduler/CassandraSchedulerTest.java,CassandraSchedulerTest,testRegistered,#,187
Before Change
@Test
public void testRegistered() throws Exception {
scheduler.registered(driver, frameworkId, masterInfo);
final Protos.FrameworkID frameworkID = stateStore.fetchFrameworkId();
assertEquals(frameworkID, this.frameworkId);
final Phase currentPhase = stageManager.getCurrentPhase();
assertTrue(currentPhase instanceof ReconciliationPhase);
After Change
@Test
public void testRegistered() throws Exception {
scheduler.registered(driver, frameworkId, masterInfo);
final Protos.FrameworkID frameworkID = stateStore.fetchFrameworkId().get();
assertEquals(frameworkID, this.frameworkId);
final Phase currentPhase = planManager.getCurrentPhase().get();
assertTrue(currentPhase instanceof ReconciliationPhase);